how to create a function with a variable number of arguments in python